How to Resolve NameError: 'train_data_flat_t' is not defined in Python